import pandas as pd
import matplotlib.pyplot as plt

# 读取CSV文件
data = pd.read_csv("cleaned_data.csv")

# 统计不同workingday对应的count总和
workingday_counts = data.groupby('workingday')['count'].sum()

# 设置中文字体
plt.rcParams['font.sans-serif'] = ['SimHei']  # 设置中文显示
plt.rcParams['axes.unicode_minus'] = False    # 解决保存图像是负号'-'显示为方块的问题

# 绘制条形图
bars = plt.bar(workingday_counts.index, workingday_counts.values)
plt.xlabel('工作日', fontsize=14)  # 修改 x 轴标签大小为 14
plt.ylabel('租用数', fontsize=14)  # 修改 y 轴标签大小为 14
plt.title('工作日租用单车总数')

# 在每个条形上方显示具体数值
for bar in bars:
    yval = bar.get_height()
    plt.text(bar.get_x() + bar.get_width()/2.0, yval, int(yval), va='bottom', ha='center')  # va='bottom'使得数字在柱子顶部显示

plt.xticks(workingday_counts.index, ['非工作日', '工作日'])  # 将数字标签改为对应的工作日
plt.show()
